【参数篇】传参

函数

• tf.flags()
• tf.app.flags()

功能

同python中的argparser作用相同，在python执行命令行时外部传入参数设置

函数

• tf.app.flags.DEFINE_string(flag__name, default_value, desc)_

• tf.app.flags.DEFINE_integer(flag__name, default_value, desc)_

• tf.app.flags.DEFINE_float(flag__name, default_value, desc)_

• tf.app.flags.DEFINE_boolean(flag__name, default_value, desc)_

使用

# test.py
import tensorflow as tf

tf.app.flags.DEFINE_string('ss', 'aa', 'test1')
tf.app.flags.DEFINE_integer('ii', 1, 'test2')
tf.app.flags.DEFINE_float('ff', 0.1, 'test3')

FLAGS = tf.flags.FLAGS
# 解析参数，直接将其转化成 flags library
## old version
FLAGS._parse_flags()      # old version
for key, value in FLAGS.__flags.items():
    print('key:{}\tvalue:{}'.format(key, value))

FLAGS.flag_values_dict()  # new version
for key, value in Flags:
    print('key:{}\tvalue:{}'.format(key, value))


# 执行
python test.py

ss:aa    ii:1    ff:0.1

python test.py --ss 'bb' --ii 2 --ff 0

ss:bb    ii:2    ff:0.2